Python program to create a directory